python code for chat gpt